from django.contrib import admin
from flybotServer.models import BotApiKey, BotInfo, BotMessage, BotActions, MJTasks, SensitiveWord

@admin.register(BotApiKey)
class ChatUsageRecorderAdmin(admin.ModelAdmin):
    list_display = ['api_key', 'level', 'state', 'user_id']

@admin.register(BotInfo)
class ChatUsageRecorderAdmin(admin.ModelAdmin):
    list_display = ['api_key', 'bot_type', 'hostname', 'username', 'password', 'bot_path', 'process_id']

@admin.register(BotMessage)
class ChatUsageRecorderAdmin(admin.ModelAdmin):
    list_display = ['bot_info', 'source', 'state', 'message']

@admin.register(BotActions)
class ChatUsageRecorderAdmin(admin.ModelAdmin):
    list_display = ['bot_info', 'action_name', 'action_content', 'status']

@admin.register(MJTasks)
class ChatUsageRecorderAdmin(admin.ModelAdmin):
    list_display = ['bot_info', 'status', 'origin_prompt', 'from_nickname', 'origin_image_url']

@admin.register(SensitiveWord)
class ChatUsageRecorderAdmin(admin.ModelAdmin):
    list_display = ['word']
